package leetcode_ago.bin_tree;

/**
 * 给定两个整数数组preorder 和 inorder
 * 其中preorder 是二叉树的先序遍历， inorder是同一棵树的中序遍历
 * 请构造二叉树并返回其根节点
 * 思路：
 * 1.不断从前序结果集中取出元素，此元素作为当前数的根节点
 * 2.拿着根节点去中序遍历中查找该节点值所处的位置，数组左侧就是左子树，右侧就是右子树
 * 3.递归上述流程，直到前序遍历结果集全部遍历结束
 */
public class Num105_BuildTreeByPreAndIn {
    //当前处理前序结果集的哪个节点
    private int index=0;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
//        TreeNode深度范围在1-3000之间，不用判空
        return buildTreeHelper(preorder,inorder,0,preorder.length-1);
    }

    private TreeNode buildTreeHelper(int[] preorder, int[] inorder, int left, int right) {
        //1.边界
        if(left>right){
            return null;
        }
        if(index==preorder.length){
            return null;
        }
        //2.构建根节点
        TreeNode root=new TreeNode(preorder[index]);
        index++;
        //3.找到该节点在中序遍历中的索引位置
        int pos=find(root.val,inorder);
        //4.根据索引位置知道树的左孩子右孩子在哪
        root.left=buildTreeHelper(preorder,inorder,left,pos-1);
        root.right=buildTreeHelper(preorder,inorder,pos+1,right);
        return root;
    }

    private int find(int val, int[] inorder) {
        for (int i = 0; i < inorder.length; i++) {
            if(inorder[i]==val){
                return i;
            }
        }
        return -1;
    }
}
